# session.py
# 创建数据库 session 和依赖注入（如FastAPI Depends）
from sqlalchemy import create_engine
from sqlalchemy.orm import sessionmaker
from app.core.config import settings # 导入配置

# 使用从 .env 文件中读取到的 DATABASE_URL
engine = create_engine(
    settings.DATABASE_URL,
    # connect_args 是 SQLite 特有的，为了允许多线程访问
    connect_args={"check_same_thread": False}
)
SessionLocal = sessionmaker(autocommit=False, autoflush=False, bind=engine)

# ... get_db 函数 ...

def get_db():
    db = SessionLocal()
    try:
        yield db
    finally:
        db.close()